#!/bin/bash

# 批量创建用户
create_users() {
    echo "请输入用户名的前缀（例如：stu）："
    read prefix
    echo "请输入要创建的用户数量："
    read count
    success_count=0
    for ((i=1; i<=count; i++))
    do
        username="${prefix}${i}"
        # 创建用户
        useradd $username > /dev/null 2>&1
        if [ $? -eq 0 ]; then
            ((success_count++))
        fi
    done
    echo "$success_count 个用户已成功创建！"
}

# 批量删除用户
delete_users() {
    echo "请输入要删除的用户名前缀（例如：stu）："
    read prefix
    echo "请输入要删除的用户数量："
    read count
    success_count=0
    for ((i=1; i<=count; i++))
    do
        username="${prefix}${i}"
        # 删除用户
        userdel $username > /dev/null 2>&1
        if [ $? -eq 0 ]; then
            ((success_count++))
        fi
    done
    echo "$success_count 个用户已成功删除！"
}

# 主菜单
while true; do
    echo "*******************************************************"
    echo "*                  用户管理脚本                    *"
    echo "*   1. 批量创建用户                                *"
    echo "*   2. 批量删除用户                                *"
    echo "*   3. 退出                                         *"
    echo "*******************************************************"
    echo "请输入选择："
    read choice
    case $choice in
        1)
            create_users
            ;;
        2)
            delete_users
            ;;
        3)
            echo "退出程序..."
            exit 0
            ;;
        *)
            echo "无效选择，请重新选择。"
            ;;
    esac
done
